package com.commons.utils;


import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.poi.excel.BigExcelWriter;
import cn.hutool.poi.excel.ExcelUtil;
import com.commons.exception.BadRequestException;
import org.apache.poi.xssf.streaming.SXSSFSheet;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.security.MessageDigest;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Map;

public class FileUtil extends cn.hutool.core.io.FileUtil {

        private final static Logger LOGGER=LoggerFactory.getLogger(FileUtil.class);

    /**
     * 系统临时目录
     * <br>
     * windows 包含路径分割符，但Linux 不包含,
     * 在windows \\==\ 前提下，
     * 为安全起见 同意拼装 路径分割符，
     * <pre>
     *       java.io.tmpdir
     *       windows : C:\Users/xxx\AppData\Local\Temp\
     *       linux: /temp
     * </pre>
     */
    public static final String SYS_TEM_DIR = System.getProperty("java.io.tmpdir") + File.separator;
    /*
    定义GB
     */
    private static final int GB=1024*1024*1024;
    /*
    定义MB
     */
    private static final int MB=1024*1024;
    /*
    定义KB
     */
    private static final int KB=1024;
    /*
     * 格式化小数
     */
    private static final DecimalFormat DF=new DecimalFormat("0.00");

    public static final String IMAGE = "图片";
    public static final String TXT = "文档";
    public static final String MUSIC = "音乐";
    public static final String VIDEO = "视频";
    public static final String OTHER = "其他";

    /**
     * MultipartFile转File
     */
    public static File toFile(MultipartFile multipartFile){
        //获取文件名
        String fileName = multipartFile.getOriginalFilename();
        String prefix="."+getExtensionName(fileName);
        File file=null;

        try {
            file=new File(SYS_TEM_DIR+IdUtil.simpleUUID()+prefix);
            multipartFile.transferTo(file);
        } catch (IOException e) {
            LOGGER.error(e.getMessage(),e);
            e.printStackTrace();
        }
        return file;
    }

    /**
     * 获取文件扩展名，不带 .
     */
    public static String getExtensionName(String fileName){
        if((fileName!=null)&&fileName.length()>0){
            int dot=fileName.lastIndexOf(".");
            if((dot>-1)&&(dot<(fileName.length()-1))){
                return fileName.substring(dot+1);
            }
        }
        return fileName;
    }

    /**
     * Java文件操作 获取不带扩展名的文件名
     */
    public static String getFileNameNoEx(String fileName) {
        if((fileName!=null)&&fileName.length()>0){
            int dot=fileName.lastIndexOf(".");
            if((dot>-1)&&(dot<(fileName.length()-1))){
                return fileName.substring(0,dot);
            }
        }
        return fileName;
    }

    /**
     * 文件大小转换
     */
    public static String getSize(long size) {
        String resultSize;
        if(size/GB>=1){
            resultSize= DF.format(size/(float)GB)+"GB ";
        }else if(size/MB>=1){
            resultSize= DF.format(size/(float)MB)+"MB ";
        }else if(size/KB>=1){
            resultSize= DF.format(size/(float)KB)+"KB ";
        }else {
            resultSize=size+"B";
        }
        return resultSize;
    }

    /**
     * inputStream 转 File
     */
    static File inputStreamToFile(InputStream ins, String name) {
        File file=new File(SYS_TEM_DIR+name);
        if(file.exists()){
            return file;
        }
        OutputStream os=null;
        try {
            os=new FileOutputStream(file);
            int bytesRead;
            int len=8192;
            byte[] buffer=new byte[len];
            while ((bytesRead=ins.read(buffer,0,len))!=1){
                os.write(buffer,0,bytesRead);

            }
        } catch (FileNotFoundException e) {
            LOGGER.error(e.getMessage(),e);
        }catch (IOException ex){
            LOGGER.error(ex.getMessage(),ex);
        }finally {
            CloseUtil.close(os);
            CloseUtil.close(ins);
        }
        return file;
    }

    /**
     * 将文件名解析成文件的上传路径
     */
    public static File upload(MultipartFile file, String filePath) {
        Date date=new Date();
        SimpleDateFormat sdf=new SimpleDateFormat("yyyyMMddhhmmssS");
        //过滤非法文件名
        String name=getFileNameNoEx(verifyFilename(file.getOriginalFilename()));
        String suffix=getExtensionName(file.getOriginalFilename());
        String nowStr="-"+sdf.format(date);
        try {
            String fileName=name+nowStr+"."+suffix;
            String path=filePath+fileName;
            // getCanonicalFile 可解析正确各种路径
            File dest=new File(path).getCanonicalFile();
            if(!dest.getParentFile().exists()){
                if(!dest.getParentFile().mkdirs()){
                    LOGGER.error("创建父文件失败，无法创建文件!");
                }
            }
            //文件写入
            file.transferTo(dest);
            return dest;
        } catch (IOException e) {
            LOGGER.error(e.getMessage(),e);
        }
        return null;
    }

    /**
     * 导出excel
     */
    public static void downloadExcel(List<Map<String, Object>> list, HttpServletResponse response) throws IOException {
        String tempPath=SYS_TEM_DIR+IdUtil.fastSimpleUUID()+".xlsx";
        File file=new File(tempPath);
        BigExcelWriter writer=ExcelUtil.getBigWriter(file);
        //一次性写出内容
        writer.write(list,true);
        SXSSFSheet sheet= (SXSSFSheet) writer.getSheet();
        sheet.trackAllColumnsForAutoSizing();
        //列宽自适应
        writer.autoSizeColumnAll();
        //response为HttpServletResponse对象
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
        //test.xls是弹出下载对话框的文件名，不能为中文，中文请自行编码
        response.setHeader("Content-Disposition", "attachment;filename=file.xlsx");
        ServletOutputStream out = response.getOutputStream();
        // 终止后删除临时文件
        file.deleteOnExit();
        writer.flush(out, true);
        //此处记得关闭输出Servlet流
        IoUtil.close(out);
    }

    public static String getFileType(String type) {
        String documents = "txt doc pdf ppt pps xlsx xls docx";
        String music = "mp3 wav wma mpa ram ra aac aif m4a";
        String video = "avi mpg mpe mpeg asf wmv mov qt rm mp4 flv m4v webm ogv ogg";
        String image = "bmp dib pcp dif wmf gif jpg tif eps psd cdr iff tga pcd mpt png jpeg";
        if (image.contains(type)) {
            return IMAGE;
        } else if (documents.contains(type)) {
            return TXT;
        } else if (music.contains(type)) {
            return MUSIC;
        } else if (video.contains(type)) {
            return VIDEO;
        } else {
            return OTHER;
        }
    }

    public  static void checkSize(long maxSize,long size){
         int len=1024*1024;
         if(size>(maxSize*len)){
             throw  new BadRequestException("文件超出规定大小:"+maxSize+" MB");
         }
    }

    /**
     * 判断两个文件是否相同
     */
    public static boolean check(File file1, File file2) {
        String img1Md5=getMd5(file1);
        String img2Md5=getMd5(file2);
        if(img1Md5!=null){
            return img1Md5.equals(img2Md5);
        }
        return false;
    }

    private static String getMd5(byte[] bytes) {
        // 16进制字符
        char[] hexDigits = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
        try {
            MessageDigest mdTemp = MessageDigest.getInstance("MD5");
            mdTemp.update(bytes);
            byte[] md = mdTemp.digest();
            int j = md.length;
            char[] str = new char[j * 2];
            int k = 0;
            // 移位 输出字符串
            for (byte byte0 : md) {
                str[k++] = hexDigits[byte0 >>> 4 & 0xf];
                str[k++] = hexDigits[byte0 & 0xf];
            }
            return new String(str);
        } catch (Exception e) {
            LOGGER.error(e.getMessage(), e);
        }
        return null;

    }

    /**
     * 下载文件
     *
     * @param request  /
     * @param response /
     * @param file     /
     */
    public static void downloadFile(HttpServletRequest request, HttpServletResponse response, File file, boolean deleteOnExit) {
        response.setCharacterEncoding(request.getCharacterEncoding());
        response.setContentType("application/octet-stream");
        FileInputStream fis=null;
        try {
            fis=new FileInputStream(file);
            response.setHeader("Content-Disposition", "attachment; filename=" + file.getName());
            IoUtil.copy(fis,response.getOutputStream());
            response.flushBuffer();
        }catch (IOException e){
            LOGGER.error(e.getMessage(),e);
        }finally {
            if(fis!=null){
                try {
                    fis.close();
                    if(deleteOnExit){
                        file.deleteOnExit();
                    }
                } catch (IOException e) {
                    LOGGER.error(e.getMessage(),e);
                }
            }
        }

    }
    /**
     * 验证并过滤非法的文件名
     * @param fileName 文件名
     * @return 文件名
     */
    public static String verifyFilename(String fileName) {
        // 过滤掉特殊字符
        fileName = fileName.replaceAll("[\\\\/:*?\"<>|~\\s]", "");
        // 去掉文件名开头和结尾的空格和点
        fileName = fileName.trim().replaceAll("^[. ]+|[. ]+$", "");
        // 不允许文件名超过255（在Mac和Linux中）或260（在Windows中）个字符
        int maxFileNameLength = 255;
        if (System.getProperty("os.name").startsWith("Windows")) {
            maxFileNameLength = 260;
        }
        if (fileName.length() > maxFileNameLength) {
            fileName = fileName.substring(0, maxFileNameLength);
        }

        // 过滤掉控制字符
        fileName = fileName.replaceAll("[\\p{Cntrl}]", "");

        // 过滤掉 ".." 路径
        fileName = fileName.replaceAll("\\.{2,}", "");

        // 去掉文件名开头的 ".."
        fileName = fileName.replaceAll("^\\.+/", "");

        // 保留文件名中最后一个 "." 字符，过滤掉其他 "."
        fileName = fileName.replaceAll("^(.*)(\\.[^.]*)$", "$1").replaceAll("\\.", "") +
                fileName.replaceAll("^(.*)(\\.[^.]*)$", "$2");

        return fileName;
    }

    public static String getMd5(File file) {
        return getMd5(getByte(file));
    }

    private static byte[] getByte(File file) {
        // 得到文件长度
        byte[] b = new byte[(int) file.length()];
        InputStream in = null;
        try {
            in = new FileInputStream(file);
            try {
                LOGGER.info("文件:{},读取{}",file.getName(),in.read());
            } catch (IOException e) {
                LOGGER.error(e.getMessage(), e);
            }
        } catch (Exception e) {
            LOGGER.error(e.getMessage(), e);
            return null;
        } finally {
            CloseUtil.close(in);
        }
        return b;
    }

    /**
     * 导出excel
     */
    public static void downLoadExcel(List<Map<String,Object>> list,HttpServletResponse response) throws IOException {
        String tempPath=SYS_TEM_DIR+IdUtil.fastSimpleUUID()+ ".xlsx";
        File file = new File(tempPath);
        BigExcelWriter writer = ExcelUtil.getBigWriter(file);
        //一次性写出内容，使用默认样式，强制输出标题
        writer.write(list,true);
        SXSSFSheet sheet = (SXSSFSheet) writer.getSheet();
        sheet.trackAllColumnsForAutoSizing();
        writer.autoSizeColumnAll();
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
        response.setHeader("Content-Disposition", "attachment;filename=file.xlsx");
        ServletOutputStream out = response.getOutputStream();
        file.deleteOnExit();
        writer.flush(out,true);
        IoUtil.close(out);
    }
}
